#pragma once
#include "iostream"
#include "opencv2/opencv.hpp"
class ArmorDetector{

public:
	ArmorDetector();
	cv::Mat src;
	
	std::vector<cv::RotatedRect> detection();
	cv::Mat BGR;
	cv::Mat HSV;
	cv::Mat HSV_img(cv::Mat &img);
	cv::Mat BGR_img(cv::Mat &img);

private:
	cv::Mat imgBinary,erosion,imgHSV;
	std::vector<cv::Mat> channels;
	std::vector<std::vector<cv::Point>> light_contours;
	double maxValue{};
	std::vector<cv::RotatedRect> ellipse;
	std::vector<cv::RotatedRect> armor_info;
	cv::Mat ProcessedImg;
	float ratio;
	
	std::vector<std::vector<cv::Point>> contourProcess();
	
	void InfoParse();
	
	
};


